#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@File:modifytime.py
@Author:Josco
@Date:2022/09/28 11:01:58
@Version:1.0
@Description: 修改文件时间的类
'''

from win32file import CreateFile, SetFileTime, GetFileTime, CloseHandle
from win32file import GENERIC_READ, GENERIC_WRITE, OPEN_EXISTING
from pywintypes import Time  # 可以忽视这个 Time 报错（运行程序还是没问题的）
import time
import os

def modifyFileTime(filePath, createTime, modifyTime, accessTime, offset):
    """
    用来修改任意文件的相关时间属性，时间格式：YYYY-MM-DD HH:MM:SS 例如：2019-02-02 00:01:02
    :param filePath: 文件路径名
    :param createTime: 创建时间
    :param modifyTime: 修改时间
    :param accessTime: 访问时间
    :param offset: 时间偏移的秒数,tuple格式，顺序和参数时间对应
    """
    try:
        format = "%Y-%m-%d %H:%M:%S"  # 时间格式
        cTime_t = timeOffsetAndStruct(createTime, format, offset[0])
        mTime_t = timeOffsetAndStruct(modifyTime, format, offset[1])
        aTime_t = timeOffsetAndStruct(accessTime, format, offset[2])

        fh = CreateFile(filePath, GENERIC_READ | GENERIC_WRITE, 0, None, OPEN_EXISTING, 0, 0)
        createTimes, accessTimes, modifyTimes = GetFileTime(fh)

        createTimes = Time(time.mktime(cTime_t))
        accessTimes = Time(time.mktime(aTime_t))
        modifyTimes = Time(time.mktime(mTime_t))
        SetFileTime(fh, createTimes, accessTimes, modifyTimes)
        CloseHandle(fh)
        return 0
    except:
        return 1

def timeOffsetAndStruct(times, format, offset):
    return time.localtime(time.mktime(time.strptime(times, format)) + offset)

def formatTime(atime): 
    return time.strftime("%Y-%m-%d %H:%M:%S",time.localtime(atime))

if __name__ == '__main__':
    # 需要自己配置
    fileinfo = os.stat("./data/11.pdf")
    cTime = formatTime(fileinfo.st_ctime)  # 创建时间
    mTime = formatTime(fileinfo.st_mtime)  # 修改时间
    aTime = formatTime(fileinfo.st_mtime)  # 访问时间
    print(cTime)
    print(mTime)
    print(aTime)
    fName = './data/1111.pdf'  # 文件路径，文件存在才能成功（可以写绝对路径，也可以写相对路径）

    offset = (0, 1, 2)  # 偏移的秒数（不知道干啥的）

    # 调用函数修改文件创建时间，并判断是否修改成功
    r = modifyFileTime(fName, cTime, mTime, aTime, offset)
    if r == 0:
        print('修改完成')
    elif r == 1:
        print('修改失败')

